#ifndef __YOWO_PLUGIN_H__
#define __YOWO_PLUGIN_H__

#include "boxInfo.h"

#ifdef __cplusplus
extern "C" {
#endif

typedef struct {
  int img_width;
  int img_height;

  float conf_threshold; // 0.1
  float nms_threshold;  // 0.5
  float vis_threshold;  // 0.3

  int model_input_h;
  int model_input_w;

  int model_output_len;

  lynBoxesInfo *boxesInfo; // 输出box信息, Lynxi plugin方式时需要使用device
                           // memory传递，否则使用系统内存
  void *output_tensor;

} YowoPostInfo_t;

int lynYowoPostProcess(YowoPostInfo_t *post_info);

#ifdef __cplusplus
}
#endif

#endif